package com.liuzh.kaoyanspringboot.service.impl;

import cn.hutool.log.Log;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.liuzh.kaoyanspringboot.common.Constants;
import com.liuzh.kaoyanspringboot.entity.Menu;
import com.liuzh.kaoyanspringboot.entity.RoleMenu;
import com.liuzh.kaoyanspringboot.exception.ServiceException;
import com.liuzh.kaoyanspringboot.mapper.MenuMapper;
import com.liuzh.kaoyanspringboot.mapper.RoleMenuMapper;
import com.liuzh.kaoyanspringboot.service.MenuService;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.List;
import java.util.stream.Collectors;


/**
 * <p>
 *  服务实现类
 * </p>
 *
 * @author 刘志鸿
 * @since 2023-03-02
 */
@Service
public class MenuServiceImpl extends ServiceImpl<MenuMapper, Menu> implements MenuService {
    private static final Log LOG = Log.get();
    @Resource
    private MenuMapper menuMapper;
    @Resource
    private RoleMenuMapper roleMenuMapper;

    @Override
    public List<Menu> findMenus() {
        QueryWrapper<Menu> queryWrapper = new QueryWrapper<>();
        // 查询所有数据
        List<Menu> list = list(queryWrapper);
        // 找出 pid 为 null 的一级菜单
        List<Menu> parentNodes = list.stream().filter(menu -> menu.getPid() == null).collect(Collectors.toList());
        // 找出一级菜单的子菜单
        for (Menu menu : parentNodes) {
            // 从 list 的一个 m 中根据pid 匹配 parentList 的数据 menu 中的 id,收集 list 中各个匹配的 m,
            // 最后放入 menu 的 children 属性中
            // 时间复杂度 m*n
            menu.setChildren(list.stream().filter(m -> menu.getId().equals(m.getPid())).collect(Collectors.toList()));
        }
        return parentNodes;
    }

    @Override
    public boolean deleteById(Integer MenuId) {
        QueryWrapper<RoleMenu> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("menu_id",MenuId);
        // 根据 MenuId 删除 roleMenu关系
        try {
            roleMenuMapper.delete(queryWrapper);
            // 再根据 MenuId 删除 菜单
            if(menuMapper.deleteById(MenuId) > 0) {
                return true;
            }
            return false;
        } catch (Exception e) {
            LOG.error(e);
            throw new ServiceException(Constants.CODE_500, "删除错误");
        }
    }
}
